package edu.yngsxy.wkstudy.core.security;

import cn.dev33.satoken.exception.*;
import edu.yngsxy.wkstudy.core.model.common.Result;
import lombok.extern.slf4j.Slf4j;
import org.springframework.core.annotation.Order;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

@RestControllerAdvice
// 数值越小，优先级越高,防止被全局异常处理器的兜底处理
@Order(1)
@Slf4j // 自动生成log日志对象
public class SaExceptionHandler {
    // 拦截：未登录异常
    @ExceptionHandler(NotLoginException.class)
    public Result<?> handlerException(NotLoginException e) {
        // 打印堆栈，以供调试
        log.error("未登录异常：", e);
        // 返回给前端
        return Result.unauthorized(e.getMessage());
    }

    // 拦截：缺少权限异常
    @ExceptionHandler(NotPermissionException.class)
    public Result<?> handlerException(NotPermissionException e) {
        log.error("缺少权限异常：", e);
        return Result.forbidden("缺少权限：" + e.getPermission());
    }

    // 拦截：缺少角色异常
    @ExceptionHandler(NotRoleException.class)
    public Result<?> handlerException(NotRoleException e) {
        log.error("缺少角色异常：", e);
        return Result.forbidden("缺少角色：" + e.getRole());
    }

    // 拦截：二级认证校验失败异常
    @ExceptionHandler(NotSafeException.class)
    public Result<?> handlerException(NotSafeException e) {
        log.error("二级认证校验失败异常：", e);
        return Result.unauthorized("二级认证校验失败：" + e.getService());
    }

    // 拦截：服务封禁异常
    @ExceptionHandler(DisableServiceException.class)
    public Result<?> handlerException(DisableServiceException e) {
        log.error("服务封禁异常：", e);
        return Result.unauthorized("当前账号 " + e.getService() + " 服务已被封禁 (level=" + e.getLevel() + ")：" + e.getDisableTime() + "秒后解封");
    }

    // 拦截：Http Basic 校验失败异常
    @ExceptionHandler(NotHttpBasicAuthException.class)
    public Result<?> handlerException(NotHttpBasicAuthException e) {
        log.error("Http Basic 校验失败异常：", e);
        return Result.error(e.getMessage());
    }
}
